#pragma once

#include <QtGui/QWidget>

#include "GeneratedFiles/ui_trainnetworkwizard.h"

class ImageLoader;
class CNetworkManager;
class NetworkRunner;

/*----------------------------------------------------------------------------*/

class TrainNetworkWizard
	:	public QWidget

/*----------------------------------------------------------------------------*/

{
	Q_OBJECT

/*----------------------------------------------------------------------------*/

public:

/*----------------------------------------------------------------------------*/

	TrainNetworkWizard( ImageLoader& _imageLoader, CNetworkManager& _networkManager );

	~TrainNetworkWizard();	

	void ShowCentered( QWidget& _parent );

/*----------------------------------------------------------------------------*/

private:

/*----------------------------------------------------------------------------*/

	Ui::TrainNetworkWizardUi* m_ui;

	ImageLoader& m_imageLoader;

	CNetworkManager& m_networkManager;

	NetworkRunner* m_networkRunner;

/*----------------------------------------------------------------------------*/

	void TrainingState( bool _state );

/*----------------------------------------------------------------------------*/

private slots:

/*----------------------------------------------------------------------------*/

	void StartTraining();

	void SelectLocation();

	void LocationEnabled( int _state );

	void TrainingFinished();

/*----------------------------------------------------------------------------*/

};

/*----------------------------------------------------------------------------*/
